import torch
from ops_test_module import OpsTest

device = 'cuda:0'
input1 = torch.tensor([[0, 1, 2], [3, 4, 5]]).to(torch.float32).to(device)
input2 = torch.tensor([[100, 100, 100], [100, 100, 100]]).to(torch.float32).to(device)

output = OpsTest.apply(input1, input2)

print(output)
pass
